package cn.tedu.v4;

import java.io.*;
import java.net.BindException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

/**
 *  Version  4
 *  多人聊天室 ---消息转发：消息广播
 *  替代 客户端也可以写多线程并发
 *  消息广播：多个客户端链接上服务器，一个人说话，其他都看到
 *
 *  功能：客户端之间消息转发
 *  1.服务器端创建集合，存储所有的客户端输出流
 *  2.广播消息给所有客户端  broadcastMessage 广播留言
 *    线程上面：ClientHandler
 *  3.将所有的客户端输出流添加  集合中---同步锁
 *
 *  ClientHandler---构造器中，目的：多线程创建成功，立马附加消息转发功能
 *   构造器 三个函数：①通过拨通服务器的socket，赋值当前socket
 *                  明确是链接成功的客户端通讯
 *                 ②拨通服务器的socket的IP地址
 *                 ③拨通服务器的socket的IP地址的同时赋予读写功能
 *                  发送客户端的数据给其他客户端【消息转发】
 *  4.收到的广播消息转发至所有客户端---broadcastMessage  广播留言
 *  添加：ClientHandle---run(),进行读写功能
 *    读入       写出
 *  5.客户端断开，需要从集合中移除:一个客户端断开，将这个客户端移除
 *  添加：ClientHandler---run()---finally块中【一定把客户端移除】
 *
 */
public class Server {
    private ServerSocket serverSocket;
    private static final Integer PORT=9086;
    //1.存储客户端的输出流：使用集合
    private List<PrintWriter> allOut=new ArrayList<>();
    //解释：用来存储所有客户端的输出流，每当有一个新的客户端链接服务器时候
    //它的输出流[PrintWriter] 会被添加至allOut 中
    //这样服务器就能向所有的客户端发送消息

    public Server(){
        try {
            serverSocket=new ServerSocket(PORT);
        }catch (BindException e){
            System.out.println("请检查端口号，被占用~~");
        }
        catch (IOException e) {
            e.printStackTrace();
        }
    }
    public void start(){
        try {
            while(true) {
                System.out.println("等待客户端连接");
                Socket socket = serverSocket.accept();
                new Thread(new ClientHandler(socket)).start();
            }
        } catch (IOException e) {
            System.out.println("客户端连接断开连接~~~");
        }finally {
            if(serverSocket!=null&&serverSocket.isClosed()){
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    System.out.println("服务器关闭失败~~~~");
                }
                System.out.println("服务器关闭成功~~Over~~~");
            }
        }
    }
    public static void main(String[] args){
        new Server().start();
    }
    //2.广播消息给所有客户端 [同步方法]  广播留言
    //同步方法：【链接多个客户端：并发】出现多条线程：
    //         一个客户端执行完毕,在执行下客户端
    // 参数：String message   客户端打印文字内容
    private synchronized void broadcastMessage(String message){
        //allOut  是集合---遍历：增强for,泛型：PrintWriter
        for (PrintWriter pw:allOut){
            pw.println("消息广播："+message);
        }
    }

    private class ClientHandler implements Runnable {
        private Socket socket;
        private String ip;

        private PrintWriter pw ;
        private BufferedReader br;

        public ClientHandler(Socket s){
            this.socket=s;
            ip=socket.getInetAddress().getHostAddress();
            try {
                br = new BufferedReader(new InputStreamReader(socket.getInputStream(), "UTF8"));
                pw=new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(),"UTF8")),true);
                //3.具备功能：将我们客户端输出流添加到  allOut集合中----同步锁
                //ClientHandler多线程类：处理多个客户端---多线程
                //拨号链接---并发出安全问题---加同步锁，目的：来一个客户端锁定
                //只给一个客户端加消息转发，同步锁监视：allOut接收一个客户端
                 synchronized (allOut){//监视器对象:集合很多个客户端
                     allOut.add(pw);
                      }
              } catch (IOException e) {
                System.out.println("客户端断开连接了~~~~");
              }
        }
        public void run() {
            try {

                new Thread(()->{
                    try {
                        String message;
                        while ((message=br.readLine())!=null) {//读入客户端
                            System.out.println(ip+"客户端说："+ message);

                            System.out.print("服务器请输入：");
                            //4.服务器端收到的一个链接成功客户端的信息转发给其他客户端
                            broadcastMessage(message);
                        }
                    } catch (IOException e) {
                        System.out.println("客户端发来的消息断开了~~~");
                    }
                }).start();

                Scanner input=new Scanner(System.in);
                while(true){
                    System.out.println("服务器输入：");
                    String sme= input.nextLine();
                    pw.println("服务器说："+sme);
                    System.out.println("写出完毕");
                }

            } finally {
                //5.客户端断开，需要从集合中移除，加同步锁
                //为啥加同步锁：保证移除对象就是转发消息方法的同步
                //理解：链接服务器端的客户端不是当前的
                synchronized (allOut){
                    allOut.remove(pw);
                }
                if(pw!=null)
                    pw.close();
                if(br!= null)
                    try {
                        br.close();
                    } catch (IOException e) {
                        System.out.println("读入流断开连接~~");
                    }

                if (socket!=null&&!socket.isClosed()){
                    System.out.println(ip+"客户端退出服务链接~~~");
                    try {
                        socket.close();
                    } catch (IOException e) {
                        System.out.println("一个客户端退出服务器链接~~");
                    }
                }
            }
        }
    }

}
